# Copyright (c) 2013, Frappe Technologies Pvt. Ltd. and contributors
# For license information, please see license.txt


from frappe import _
from frappe.utils import flt

import erpnext
from erpnext.loan_management.report.applicant_wise_loan_security_exposure.applicant_wise_loan_security_exposure import (
	get_applicant_wise_total_loan_security_qty,
	get_loan_security_details,
)


def execute(filters=None):
	columns = get_columns(filters)
	data = get_data(filters)
	return columns, data

def get_columns(filters):
	columns = [
		{"label": _("Loan Security"), "fieldname": "loan_security", "fieldtype": "Link", "options": "Loan Security", "width": 160},
		{"label": _("Loan Security Code"), "fieldname": "loan_security_code", "fieldtype": "Data", "width": 100},
		{"label": _("Loan Security Name"), "fieldname": "loan_security_name", "fieldtype": "Data", "width": 150},
		{"label": _("Haircut"), "fieldname": "haircut", "fieldtype": "Percent", "width": 100},
		{"label": _("Loan Security Type"), "fieldname": "loan_security_type", "fieldtype": "Link", "options": "Loan Security Type", "width": 120},
		{"label": _("Disabled"), "fieldname": "disabled", "fieldtype": "Check", "width": 80},
		{"label": _("Total Qty"), "fieldname": "total_qty", "fieldtype": "Float", "width": 100},
		{"label": _("Latest Price"), "fieldname": "latest_price", "fieldtype": "Currency", "options": "currency", "width": 100},
		{"label": _("Price Valid Upto"), "fieldname": "price_valid_upto", "fieldtype": "Datetime", "width": 100},
		{"label": _("Current Value"), "fieldname": "current_value", "fieldtype": "Currency", "options": "currency", "width": 100},
		{"label": _("% Of Total Portfolio"), "fieldname": "portfolio_percent", "fieldtype": "Percentage", "width": 100},
		{"label": _("Pledged Applicant Count"), "fieldname": "pledged_applicant_count", "fieldtype": "Percentage", "width": 100},
		{"label": _("Currency"), "fieldname": "currency", "fieldtype": "Currency", "options": "Currency", "hidden": 1, "width": 100},
	]

	return columns

def get_data(filters):
	data = []
	loan_security_details = get_loan_security_details()
	current_pledges, total_portfolio_value = get_company_wise_loan_security_details(filters, loan_security_details)
	currency = erpnext.get_company_currency(filters.get('company'))

	for security, value in current_pledges.items():
		if value.get('qty'):
			row = {}
			current_value = flt(value.get('qty', 0) * loan_security_details.get(security, {}).get('latest_price', 0))
			valid_upto = loan_security_details.get(security, {}).get('valid_upto')

			row.update(loan_security_details.get(security))
			row.update({
				'total_qty': value.get('qty'),
				'current_value': current_value,
				'price_valid_upto': valid_upto,
				'portfolio_percent': flt(current_value * 100 / total_portfolio_value, 2),
				'pledged_applicant_count': value.get('applicant_count'),
				'currency': currency
			})

			data.append(row)

	return data


def get_company_wise_loan_security_details(filters, loan_security_details):
	pledge_values, total_value_map, applicant_type_map = get_applicant_wise_total_loan_security_qty(filters,
		loan_security_details)

	total_portfolio_value = 0
	security_wise_map = {}
	for key, qty in pledge_values.items():
		security_wise_map.setdefault(key[1], {
			'qty': 0.0,
			'applicant_count': 0.0
		})

		security_wise_map[key[1]]['qty'] += qty
		if qty:
			security_wise_map[key[1]]['applicant_count'] += 1

		total_portfolio_value += flt(qty * loan_security_details.get(key[1], {}).get('latest_price', 0))

	return security_wise_map, total_portfolio_value
